# coding=utf-8
# 代码文件：chapter7/ch7.1.4.py

from docx import Document

document = Document('data/temp.docx')

# 表格中的数据
records = [
    ('西城', 201991),
    ('海淀', 129508),
    ('东城', 130969),
    ('朝阳', 119170),
    ('丰台', 79933),
]
# 文档后追加表格
table = document.add_table(rows=1, cols=2, style='Medium Grid 1 Accent 5')
hdr_cells = table.rows[0].cells        # 获得标表头行的所有单元格

hdr_cells[0].text = '城区'             # 设置表格的表头第1列的单元格
hdr_cells[1].text = '单价（元）'        # 设置表格的表头第2列的单元格

for addr_dist, price in records:      # 遍历表格所有行
    row_cells = table.add_row().cells # 获得一行中所有单元格
    row_cells[0].text = addr_dist     # 设置第1列的单元格
    row_cells[1].text = str(price)    # 设置第2列的单元格

document.save('data/temp.docx')

print("保存文档成功.")
